def power(x: int):
    i = 1
    while i < x:
        # print(i ** 2)
        yield i ** 2
        i += 1
    return 'done'


g = power(7)
print(g)
for i in g:
    print(i)
